import java.util.HashMap;
import java.util.Map;

class Solution {
    public int subarraysDivByK1(int[] nums, int k) {
        // sub_sum % k == 0
        // 暴力枚举：
        int count = 0;
        for (int i = 0; i < nums.length; i++) {
            int sum = 0;
            for (int j = i; j < nums.length; j++) {
                sum += nums[j];
                if (sum % k == 0) {
                    count++;
                }
            }
        }
        return count;
    }

    public int subarraysDivByK(int[] nums, int k) {
        Map<Integer, Integer> hash = new HashMap<>();
        hash.put(0,1);
        int sum = 0;
        int count = 0;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i]; // i位置的前缀和
            // 求 sum%k == x%k 的个数
            count += hash.getOrDefault((sum%k+k)%k ,0);
            hash.put((sum%k+k)%k, hash.getOrDefault((sum%k+k)%k ,0)+1);
        }
        return count;
    }

    public static int subarraysDivByK2(int[] nums, int k) {
        Map<Integer, Integer> hash = new HashMap<>();
        hash.put(0,1);
        int sum = 0;
        int count = 0;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i]; // i位置的前缀和
            // 求 sum%k == x%k 的个数
            count += hash.getOrDefault(sum%k ,0);
            hash.put(sum%k, hash.getOrDefault(sum%k ,0)+1);
        }
        return count;
    }

    public static void main(String[] args) {
        subarraysDivByK2(new int[]{-1,2,9}, 2);
    }
}